package main;

import java.util.ArrayList;
import java.util.HashMap;




public class BaselineTagger {

	private HashMap<String, Word> words;
	private ArrayList<String[]> rawfile;

	public BaselineTagger(HashMap<String, Word> words2) {
		this.words = words2;
		this.rawfile = rawfile;
		
	}
	
	



	public ArrayList<String[]> tagWords(ArrayList<String[]> rawfile){
		ArrayList<String[]> tags = new ArrayList<String[]>();
		
		for(String [] s: rawfile){
			Word w = words.get(s[0]);
			if(w != null){
				tags.add(new String[]{s[0],s[1],w.mostFrequent()});
			}else{
				tags.add(new String[]{s[0],"#NAN","#NAN2"});
			}
			
		}
		
		return tags;
		
	}

}
